#include <iostream>
#include <pthread.h>
#include <string>
#include <vector>
#include <unistd.h>

#define NUM 5
int cnt = 10000;
pthread_mutex_t glock = PTHREAD_MUTEX_INITIALIZER; // 定义锁
pthread_cond_t gcond = PTHREAD_COND_INITIALIZER;    // 定义条件变量

// 等待之前是需要对资源进行判定的, 判定本身就要访问临界资源，
// 判定结果也一定是在临界区内部
// 所以条件不满足要休眠，一定是在临界区内部休眠
// 条件变量允许线程等待, 允许一个线程唤醒在cond等待的其他线程, 实现同步过程

void* threadrun(void* args)
{
    std::string name = static_cast<const char*>(args);
    while(true)
    {
        pthread_mutex_lock(&glock);
        // 直接让对应的线程进行等待, 临界资源不满足导致我们等待的！
        // glock在pthread_cond_wait之前, 会被自动释放掉
        pthread_cond_wait(&gcond, &glock);

        std::cout << name << " 计算cnt: " << cnt << std::endl;
        cnt++;
        pthread_mutex_unlock(&glock);
    }
    return nullptr;
}

int main()
{
    std::vector<pthread_t> threads;
    for (int i = 0; i < NUM; i++)
    {
        pthread_t tid;
        char* name = new char[64];
        snprintf(name, 64, "thread-%d", i);
        int n = pthread_create(&tid, nullptr, threadrun, name);
        if(n != 0)
            continue;
        threads.push_back(tid);
        sleep(1);
    }
    
    sleep(2); // 2秒之后开始运行程序

    // 每隔一秒唤醒一个线程
    while (true)
    {
        // std::cout << "唤醒一个线程..." << std::endl;
        // pthread_cond_signal(&gcond);

        std::cout << "唤醒所有线程..." << std::endl;
        pthread_cond_broadcast(&gcond);
        sleep(1);
    }
    

    for(auto &id : threads)
    {
        int m = pthread_join(id, nullptr);
        (void)m;
    }

    return 0;
}
